Data Structures and Algorithms using Python Week 8 Programming Assignment